package main

import (
	"admin/api"
	"admin/config"
	"admin/service"
	"context"
	"errors"
	"log"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"
)

func main() {
	// 初始化配置
	cfg := config.LoadConfig()

	// 创建服务容器
	container, err := service.NewContainer(cfg)
	if err != nil {
		log.Fatalf("初始化服务容器失败: %v", err)
	}
	defer func(container *service.Container) {
		err := container.Close()
		if err != nil {

		}
	}(container)

	// 设置路由
	r := api.InitRouter(container)

	// 创建HTTP服务器
	srv := &http.Server{
		Addr:    cfg.Server.Port,
		Handler: r,
	}

	// 在goroutine中启动服务器
	go func() {
		log.Println("正在启动服务器于端口", cfg.Server.Port)
		if err := srv.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
			log.Fatalf("启动服务器失败: %v", err)
		}
	}()

	// 等待中断信号以优雅地关闭服务器
	quit := make(chan os.Signal, 1)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
	<-quit
	log.Println("正在关闭服务器...")

	// 创建一个5秒的超时上下文用于关闭
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	// 关闭服务器
	if err := srv.Shutdown(ctx); err != nil {
		log.Fatal("服务器强制关闭:", err)
	}

	// 关闭服务容器
	if err := container.Close(); err != nil {
		log.Fatal("关闭服务容器失败:", err)
	}

	log.Println("服务器已退出")
}
